package backtrack;

import java.util.ArrayList;
import java.util.List;

/**
 * @author : Rookie
 * @description : 组合问题
 * @date : 2023/6/18
 */
public class Assembly {

    public static void main(String[] args) {
        int n = 4,k = 2;
        List<Integer> path = new ArrayList<>();
        List<List<Integer>> result = new ArrayList<>();
        backtrack(4,2,1,path,result);
        System.out.println(result);
    }

    /**
     * 回溯法
     * @param n 1-n中
     * @param k k个数的组合
     * @param startIndex 开始搜索的数
     * @param path 当前结果
     * @param result 最终结果
     */
    private static void backtrack(int n, int k, int startIndex, List<Integer> path, List<List<Integer>> result) {
        if (path.size() == k) {
            // 这里为了后续path的修改不影响之前的值，每次都是new一个arrayList
            result.add(new ArrayList<>(path));
            return;
        }

        for (int i = startIndex; i <= n ; i++) {
            // 将当前的路径加入path
            path.add(i);
            backtrack(n,k,i + 1,path,result);
            // 递归完成后，移除最后一次的选择
            path.remove(path.size() - 1);
        }
    }
}
